package com.atguigu.aiproject.service;

import com.atguigu.aiproject.entity.Img;
import com.atguigu.aiproject.entity.Result;
import com.baomidou.mybatisplus.extension.service.IService;

import java.util.List;

public interface ImgService extends IService<Img> {
    /**
     * 获取可用模型列表
     * @return 模型列表
     */
    List<String> getModelsList();

    /**
     * 创建图片（支持多张）
     * @param prompt 图片描述
     * @param num 生成数量
     * @return 包含任务ID的结果
     */
    String createImages(String prompt, int num);

    /**
     * 查询图片生成状态
     * @param taskId 任务ID
     * @return 图片URL列表或任务状态
     */
    List<String> getImageStatus(String taskId);

    /**
     * 创建单张图片
     * @param prompt 图片描述
     * @return 图片URL
     */
    String createSingleImage(String prompt);
}
